package org.qydjk.common.util.wechat;

import com.thoughtworks.xstream.XStream;
import com.thoughtworks.xstream.core.util.QuickWriter;
import com.thoughtworks.xstream.io.HierarchicalStreamWriter;
import com.thoughtworks.xstream.io.xml.PrettyPrintWriter;
import com.thoughtworks.xstream.io.xml.XppDriver;
import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.qydjk.common.vo.wechat.*;

import javax.servlet.http.HttpServletRequest;
import java.io.InputStream;
import java.io.Writer;
import java.util.*;

/**
 * 消息工具类
 */
public  class WechatMessageUtil {

	/**
	 * 返回消息类型：文本
	 */
	public static final String RESP_MESSAGE_TYPE_TEXT = "text";

	/**
	 * 返回消息类型：音乐
	 */
	public static final String RESP_MESSAGE_TYPE_MUSIC = "music";

	/**
	 * 返回消息类型：图文
	 */
	public static final String RESP_MESSAGE_TYPE_NEWS = "news";

	/**
	 * 请求消息类型：文本
	 */
	public static final String REQ_MESSAGE_TYPE_TEXT = "text";

	/**
	 * 请求消息类型：图片
	 */
	public static final String REQ_MESSAGE_TYPE_IMAGE = "image";

	/**
	 * 请求消息类型：链接
	 */
	public static final String REQ_MESSAGE_TYPE_LINK = "link";

	/**
	 * 请求消息类型：地理位置
	 */
	public static final String REQ_MESSAGE_TYPE_LOCATION = "location";

	/**
	 * 请求消息类型：音频
	 */
	public static final String REQ_MESSAGE_TYPE_VOICE = "voice";

	/**
	 * 请求消息类型：推送
	 */
	public static final String REQ_MESSAGE_TYPE_EVENT = "event";

	/**
	 * 事件类型：subscribe(订阅)或扫专属二维码订阅
	 */
	public static final String EVENT_TYPE_SUBSCRIBE = "subscribe";
	
	/**
	 * 事件类型：已经关注再次扫专属二维码订阅
	 */
	public static final String EVENT_TYPE_SCAN = "SCAN";

	/**
	 * 事件类型：unsubscribe(取消订阅)
	 */
	public static final String EVENT_TYPE_UNSUBSCRIBE = "unsubscribe";

	/**
	 * 事件类型：CLICK(自定义菜单点击事件)
	 */
	public static final String EVENT_TYPE_CLICK = "CLICK";
	/**
	 * 事件类型：LOCATION(上报地理位置)
	 */
	public static final String EVENT_TYPE_LOCATION = "LOCATION";

	/**
	 * 事件类型：scancode_push(扫码推事件的事件推送)
	 */
	public static final String EVENT_TYPE_SCANCODE_PUSH = "scancode_push";

	/**
	 * 事件类型：scancode_waitmsg(扫码推事件且弹出“消息接收中”提示框的事件推送)
	 */
	public static final String EVENT_TYPE_SCANCODE_WAITING = "scancode_waitmsg";

	/**
	 * 事件类型：location_select(弹出地理位置选择器的事件推送)
	 */
	public static final String EVENT_TYPE_LOCATION_SELECT = "location_select";


	/*-------------------------------------- 回复样式 -------------------------------------------*/
	/**
	 * 纯文本
	 */
	public static final int REPLY_TYPE_CONTENT = 0;
	/**
	 * 单图
	 */
	public static final int REPLY_TYPE_IMAGE = 1;
	/**
	 * 单图文
	 */
	public static final int REPLY_TYPE_OME_IMAGE_CONTENT = 2;
	/**
	 * 图文列表
	 */
	public static final int REPLY_TYPE_MORE_IMAGE_CONTENT= 3;

	/* -------------------------------------- 回复类型 --------------------------------------------- */
	/**
	 * 默认回复
	 */
	public static final Integer REQUST_EVENT_TYPE_DEFAULT = 0;
	/**
	 * 关键字回复
	 */
	public static final Integer REQUST_EVENT_TYPE_KEYWORD = 1;
	/**
	 * 地理位置回复
	 */
	public static final Integer REQUST_EVENT_TYPE_LOCATION = 2;
	/**
	 * 普通关注回复
	 */
	public static final Integer REQUST_EVENT_TYPE_COMMON_CONCERN = 3;
	/**
	 * 带参关注回复
	 */
	public static final Integer REQUST_EVENT_TYPE_PARAMETER_FOLLOW = 4;
	/**
	 * 按钮交互回复
	 */
	public static final Integer REQUST_EVENT_TYPE_CLICK = 5;
	/**
	 * 扫码待事件回复
	 */
	public static final Integer REQUST_EVENT_TYPE_CODE_WAIT = 6;
	/**
	 * 扫码推事件回复
	 */
	public static final Integer REQUST_EVENT_TYPE_CODE_PUSH = 7;


	/**
	 * 解析微信发来的请求（XML）
	 * 
	 * @param request
	 * @return
	 * @throws Exception
	 */
	@SuppressWarnings("unchecked")
	public static Map<String, Object> parseXml(HttpServletRequest request) throws Exception {
		// 从request中取得输入流
		InputStream inputStream = request.getInputStream();
		// 读取输入流
		SAXReader reader = new SAXReader();
		Document document = reader.read(inputStream);
		// 得到xml根元素
		Element root = document.getRootElement();
		Map<String, Object> map = buildMap(root);

		// 释放资源
		inputStream.close();
		inputStream = null;
		return map;
	}

	/**
	 * xml转map
	 * @param ele
	 * @return
	 */
	private static Map<String, Object> buildMap(Element ele){
		Map<String, Object> map = new HashMap<String, Object>();
		List<Element> elementList = ele.elements();
		// 遍历所有子节点
		for (Element e : elementList){
			if(e.elements().size() > 0){
				map.put(e.getName(),buildMap(e));
			}else{
				map.put(e.getName(), e.getText());
			}
		}
		return map;
	}

	/**
	 * 文本消息对象转换成xml
	 *
	 * @param textMessage 文本消息对象
	 * @return xml
	 */
	public static String textMessageToXml(TextMessage textMessage) {
		xstream.alias("xml", textMessage.getClass());
		return xstream.toXML(textMessage);
	}

	/**
	 * 图片消息对象转换成xml
	 *
	 * @param imageMessage 图片消息对象
	 * @return xml
	 */
	public static String imageMessageToXml(ImageMessage imageMessage) {
		xstream.alias("xml", imageMessage.getClass());
		return xstream.toXML(imageMessage);
	}

	/**
	 * 音乐消息对象转换成xml
	 * 
	 * @param musicMessage 音乐消息对象
	 * @return xml
	 */
	public static String musicMessageToXml(MusicMessage musicMessage) {
		xstream.alias("xml", musicMessage.getClass());
		return xstream.toXML(musicMessage);
	}

	/**
	 * 图文消息对象转换成xml
	 * 
	 * @param newsMessage 图文消息对象
	 * @return xml
	 */
	public static String newsMessageToXml(NewsMessage newsMessage) {
		xstream.alias("xml", newsMessage.getClass());
		xstream.alias("item", new Article().getClass());
		return xstream.toXML(newsMessage);
	}

	/**
	 * 扩展xstream，使其支持CDATA块
	 * 
	 * @date 2013-05-19
	 */
	private static XStream xstream = new XStream(new XppDriver() {
		public HierarchicalStreamWriter createWriter(Writer out) {
			return new PrettyPrintWriter(out) {
				// 对所有xml节点的转换都增加CDATA标记
				boolean cdata = true;

				@SuppressWarnings({ "unused", "rawtypes" })
				public void startNode(String name, Class clazz) {
					super.startNode(name);
				}

				protected void writeText(QuickWriter writer, String text) {
					if (cdata) {
						writer.write("<![CDATA[");
						writer.write(text);
						writer.write("]]>");
					} else {
						writer.write(text);
					}
				}
			};
		}
	});
	
	/**
	 * emoji表情转换(hex -> utf-16)
	 * 
	 * @param hexEmoji
	 * @return
	 */
	public static String emoji(int hexEmoji) {
		return String.valueOf(Character.toChars(hexEmoji));
	}
	
	/**
	 * 构建单条图文消息
	 * @param fromUserName
	 * @param toUserName
	 * @param titile
	 * @param des
	 * @param pic
	 * @param url
	 * @return
	 */
	public static String buildOneNew(String fromUserName,String toUserName,String titile,String des,String pic,String url){
		// 创建图文消息
		NewsMessage newsMessage = new NewsMessage();
		newsMessage.setToUserName(fromUserName);
		newsMessage.setFromUserName(toUserName);
		newsMessage.setCreateTime(new Date().getTime());
		newsMessage.setMsgType(WechatMessageUtil.RESP_MESSAGE_TYPE_NEWS);
		newsMessage.setFuncFlag(0);
		
		List<Article> articleList = new ArrayList<Article>();
		Article article = new Article();
		article.setTitle(titile);
		article.setDescription(des);
		article.setPicUrl(pic);
		article.setUrl(url);
		articleList.add(article);
		// 设置图文消息个数
		newsMessage.setArticleCount(articleList.size());
		// 设置图文消息包含的图文集合
		newsMessage.setArticles(articleList);
		// 将图文消息对象转换成xml字符串
		String respessage = WechatMessageUtil.newsMessageToXml(newsMessage);
		return respessage;
	}
	
	/**
	 * 构建单图文或者多图文消息
	 * @param fromUserName
	 * @param toUserName
	 * @param articleList
	 * @return
	 */
	public static String buildOneOrMoreNew(String fromUserName,String toUserName,List<Article> articleList){
		// 创建图文消息
		NewsMessage newsMessage = new NewsMessage();
		newsMessage.setToUserName(fromUserName);
		newsMessage.setFromUserName(toUserName);
		newsMessage.setCreateTime(new Date().getTime());
		newsMessage.setMsgType(WechatMessageUtil.RESP_MESSAGE_TYPE_NEWS);
		newsMessage.setFuncFlag(0);
		
		// 设置图文消息个数
		newsMessage.setArticleCount(articleList.size());
		// 设置图文消息包含的图文集合
		newsMessage.setArticles(articleList);
		// 将图文消息对象转换成xml字符串
		String respessage = WechatMessageUtil.newsMessageToXml(newsMessage);
		return respessage;
	}
}